package com.common.boot.generator;

import cn.smallbun.screw.core.Configuration;
import cn.smallbun.screw.core.engine.EngineConfig;
import cn.smallbun.screw.core.engine.EngineFileType;
import cn.smallbun.screw.core.engine.EngineTemplateType;
import cn.smallbun.screw.core.execute.DocumentationExecute;
import cn.smallbun.screw.core.process.ProcessConfig;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;

import javax.sql.DataSource;
import java.util.Collections;
import java.util.List;

/**
 * @author tanghailan
 * @date 2020-10-13 17:24
 */
public class SqlDocGenerator {
    public static void main(String[] args) {
        HikariConfig hikariConfig = new HikariConfig();
        hikariConfig.setDriverClassName("com.mysql.cj.jdbc.Driver");
    hikariConfig.setJdbcUrl(
        "jdbc:mysql://10.197.184.40:3306/hzyzt?characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=false&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=GMT%2B8&nullCatalogMeansCurrent=true&allowPublicKeyRetrieval=true");
    hikariConfig.setUsername("mysqladmin");
    hikariConfig.setPassword("Zdkx!@#gz@$2021");
        // 设置可以获取tables remarks信息
        hikariConfig.addDataSourceProperty("useInformationSchema", "true");
        hikariConfig.setMinimumIdle(2);
        hikariConfig.setMaximumPoolSize(5);
        DataSource dataSource = new HikariDataSource(hikariConfig);

    // 1、生成文件配置
    EngineConfig engineConfig =
        EngineConfig.builder()
            // 生成文件路径
            //		        .fileOutputDir("D:\\work\\doc\\数据库文档")
            .fileOutputDir("/Users/tanghailan/Desktop/codeman")
            // 打开目录
            .openOutputDir(false)
            // 文件类型
            .fileType(EngineFileType.WORD)
            // 生成模板实现
            .produceType(EngineTemplateType.freemarker)
            .build();

    List<String> designatedPrefix = Collections.singletonList("yzt_");

    // 2、配置想要获取或忽略的表
    ProcessConfig processConfig =
        ProcessConfig.builder().designatedTablePrefix(designatedPrefix).build();

        // 3、生成文档配置（包含以下自定义版本号、描述等配置连接）
        Configuration config = Configuration.builder().version("1.0.0").description("数据库文档").dataSource(dataSource)
                .engineConfig(engineConfig).produceConfig(processConfig).build();

        // 4、执行生成
        new DocumentationExecute(config).execute();
    }
}
